package com.irdstudio.apicenter.gateway.util;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Set;

import org.dom4j.Document;
import org.dom4j.DocumentException;
import org.dom4j.DocumentHelper;
import org.dom4j.Element;
import org.dom4j.Namespace;
import org.dom4j.QName;

/**
 * 通用的解析xml报文为HashMap类
 * 
 * @author lizhudong
 * @date 2015-11-4
 */
public class XmlToHashMapUtil {
	private static Object xml2map(Element element) {
		Map<String, Object> map = new HashMap<String, Object>();
		List<Element> elements = element.elements();
		if (elements.size() == 0) {
			map.put(element.getName(), element.getText());
			if (!element.isRootElement()) {
				return element.getText();
			}
		} else if (elements.size() == 1) {
			map.put(elements.get(0).getName(), xml2map(elements.get(0)));
		} else if (elements.size() > 1) {
			// 多个子节点的话就得考虑list的情况了，比如多个子节点有节点名称相同的
			// 构造一个map用来去重
			Map<String, Element> tempMap = new HashMap<String, Element>();
			for (Element ele : elements) {
				tempMap.put(ele.getName(), ele);
			}
			Set<String> keySet = tempMap.keySet();
			for (String string : keySet) {
				Namespace namespace = tempMap.get(string).getNamespace();
				List<Element> elements2 = element.elements(new QName(string,
						namespace));
				// 如果同名的数目大于1则表示要构建list
				if (elements2.size() > 1) {
					List<Object> list = new ArrayList<Object>();
					for (Element ele : elements2) {
						list.add(xml2map(ele));
					}
					map.put(string, list);
				} else {
					// 同名的数量不大于1则直接递归去
					map.put(string, xml2map(elements2.get(0)));
				}
			}
		}
		return map;
	}

	/**
	 * 传入标准的xml格式的字符串
	 * 
	 * @param xmlStr
	 * @return HashMap<String, Object>
	 */
	public static HashMap<String, Object> xml2map(String xmlStr) {
		if (xmlStr == null || "".equals(xmlStr.trim())) {
			return null;
		}
		if (null != xmlStr && !"".equals(xmlStr)) {
			if (xmlStr.indexOf("<") != -1 && xmlStr.lastIndexOf(">") != -1
					&& xmlStr.lastIndexOf(">") > xmlStr.indexOf("<"))
				xmlStr = xmlStr.substring(xmlStr.indexOf("<"),
						xmlStr.lastIndexOf(">") + 1);
		}
		try {
			Document document = DocumentHelper.parseText(xmlStr);
			return (HashMap<String, Object>) xml2map(document.getRootElement());
		} catch (DocumentException e) {
			// TODO Auto-generated catch block
			e.printStackTrace();
		}
		return null;
	}

	/**
	 * 传入标准的org.dom4j.Document对象
	 * 
	 * @param document
	 * @return HashMap<String, Object>
	 */
	public static HashMap<String, Object> xml2map(Document document) {
		if (document == null) {
			return null;
		}
		return (HashMap<String, Object>) xml2map(document.getRootElement());
	}
}
